// hello/database/database.go
package database

import (
	"fmt"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

// DB 全局数据库连接变量
var DB *gorm.DB

// InitDB 初始化数据库连接
func InitDB() {
	dsn := "root:root@tcp(127.0.0.1:3306)/demo?charset=utf8mb4&parseTime=True&loc=Local"
	var err error
	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		fmt.Printf("Failed to connect to database: %v\n", err)
		return
	}
	// 自动迁移模型
	// 这里假设你有一个 User 模型，你可以根据实际情况添加更多模型
	// 你可以在 models 包中定义 User 模型
	// 例如：package models; type User struct { gorm.Model; Name string; Address string; Age int; Status int }
	// 这里需要导入 models 包
	// import "hello/models"
	// DB.AutoMigrate(&models.User{})
}
